dictionary input in python